package com.stock.stockprediction.entity;

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import org.springframework.data.annotation.Id;
import org.springframework.data.mongodb.core.mapping.Document;
import org.springframework.data.mongodb.core.mapping.Field;

@Document("stock")
@Data
@AllArgsConstructor
@NoArgsConstructor
public class StockMongo {

    @Id
    private String id;

    //股票真实数据
    @Field
    private double[] realStock;

    //股票预测数据
    @Field
    private double[] predicateStock;

    //训练集损失函数
    @Field
    private double[] loss;

    //验证集损失函数
    @Field
    private double[] val_loss;

    //均方误差
    @Field
    private double mse;

    //均方根误差
    @Field
    private double rmse;

    //平方绝对误差
    @Field
    private double mae;

    //标签：SimpleRNN或者LSTM，区分预测算法
    @Field
    private String label;

}
